package com.wisdomparking.config.result;

import org.springframework.core.MethodParameter;
import org.springframework.http.MediaType;
import org.springframework.http.converter.HttpMessageConverter;
import org.springframework.http.server.ServerHttpRequest;
import org.springframework.http.server.ServerHttpResponse;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.servlet.mvc.method.annotation.ResponseBodyAdvice;

@ControllerAdvice
public class ResultHandle implements ResponseBodyAdvice<Object> {

	@Override
	public boolean supports(MethodParameter methodParameter, Class<? extends HttpMessageConverter<?>> aClass) {
		return true;
	}
	
	@Override
	public Object beforeBodyWrite(Object object, MethodParameter methodParameter, MediaType mediaType,
			Class<? extends HttpMessageConverter<?>> aClass, ServerHttpRequest serverHttpRequest,
			ServerHttpResponse serverHttpResponse) {
		
		if (object instanceof Result) {
			Result result = (Result) object;
			if (!result.getResCode().equals("0")) {
				return result;
			}
		}
		
		if (object instanceof ResponseData) {
			return object;
		}
		
		if (object instanceof Boolean){
			if(String.valueOf(object).equals("false")){
				return ResultUtil.error(0, "操作失败");
			}
		}
		
		return ResultUtil.success(object);
		
	}
}
